Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Unify weight handling and refactor linear models' helper functions #267

Merged
merged 8 commits into from
May 16, 2024

Conversation

JoaquinIglesiasTurina
Copy link
Contributor

Implements the suggestion from here.

I've had to revert the change suggested by @krstopro here, as that change made some tests from LinearRegression and RidgeRegression fail.

When it comes to LinearRegression I believe those failures are related to the shape of the target data:

  • The tests yield failures when the data is 2 dimensional, like a column vector
  • Flattening those data yield a passing test.

RidgeRegression tests fail with multi-output target data. These cannot be flattened.

@msluszniak
Copy link
Contributor

Thank you for the PR. I'll investigate if we can implement the suggestion, so it'll be universal.

Copy link
Contributor

@msluszniak msluszniak left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it's now good to go ;)

@josevalim josevalim merged commit 09d500a into elixir-nx:main May 16, 2024
2 checks passed
@josevalim
Copy link
Contributor

💚 💙 💜 💛 ❤️

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants